package com.ita.os.merchant.util;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

public class UploadTool {
	private HttpServletRequest request = null;
	private String savePath = "";
	private String tempPath = "";
	private List<FileBean> fileBeanList = null;
	//上传表单的其它信息，除file类型的input
	private Map<String,String> formDatas = null;
	private String uploadStatus = "";
	//文件单个最大大小
	private long maxSize;
	//文件允许的格式为
	private String allowExts;
	//文件不允许的格式为
	private String unAllowExts;
	private String message = "";


	public void uploadFiles() throws FileUploadException, IOException{
		savePath = this.getSavePath();
		tempPath = this.getTempPath();
		formDatas = new HashMap<String, String>();
		fileBeanList = new ArrayList<FileBean>();
		File tmpFile = new File(tempPath);
		if(!tmpFile.exists())
			tmpFile.mkdir();

			// 使用Apache文件上传组件处理文件上传步骤：
			// 1、创建一个DiskFileItemFactory工厂
			DiskFileItemFactory factory = new DiskFileItemFactory();
			// 设置工厂的缓冲区的大小，当上传的文件大小超过缓冲区的大小时，就会生成一个临时文件存放到指定的临时目录当中。
			factory.setSizeThreshold(1024 * 100);// 设置缓冲区的大小为100KB，如果不指定，那么缓冲区的大小默认是10KB
			// 设置上传时生成的临时文件的保存目录
			factory.setRepository(tmpFile);
			// 2、创建一个文件上传解析器
			ServletFileUpload upload = new ServletFileUpload(factory);
			// 监听文件上传进度
			upload.setProgressListener(new ProgressListener() {
				public void update(long pBytesRead, long pContentLength, int arg2) {
					System.out.println("文件大小为：" + pContentLength + ",当前已处理：" + pBytesRead);
					/**
					 * 文件大小为：14608,当前已处理：4096 文件大小为：14608,当前已处理：7367
					 * 文件大小为：14608,当前已处理：11419 文件大小为：14608,当前已处理：14608
					 */
				}
			});
			// 解决上传文件名的中文乱码
			upload.setHeaderEncoding("UTF-8");
			// 3、判断提交上来的数据是否是上传表单的数据
			if (!ServletFileUpload.isMultipartContent(this.getRequest())) {
				// 按照传统方式获取数据
				return;
			}
			// 设置上传单个文件的大小的最大值，目前是设置为1024*1024字节，也就是1MB
						upload.setFileSizeMax(1024 * 1024);
						// 设置上传文件总量的最大值，最大值=同时上传的多个文件的大小的最大值的和，目前设置为10MB
						upload.setSizeMax(1024 * 1024 * 10);
						// 4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，每一个FileItem对应一个Form表单的输入项
						List<FileItem> list = upload.parseRequest(this.getRequest());
						FileBean fileBean = null;
						for (FileItem item : list) {
							// 如果fileitem中封装的是普通输入项的数据
							if (item.isFormField()) {
								String name = item.getFieldName();
								// 解决普通输入项的数据的中文乱码问题
								String value = item.getString("UTF-8");
								// value = new String(value.getBytes("iso8859-1"),"UTF-8");
								formDatas.put(name, value);
								System.out.println(name + "=" + value);
							} else {// 如果fileitem中封装的是上传文件 
									// 得到上传的文件名称，
								String filename = item.getName();
								System.out.println(filename);
								if (filename == null || filename.trim().equals("")) {
									continue;
								}
								fileBean = new FileBean();
								// 注意：不同的浏览器提交的文件名是不一样的，有些浏览器提交上来的文件名是带有路径的，如：
								// c:\a\b\1.txt，而有些只是单纯的文件名，如：1.txt
								// 处理获取到的上传文件的文件名的路径部分，只保留文件名部分
								filename = filename.substring(filename.lastIndexOf("\\") + 1);
								// 得到上传文件的扩展名
								String fileExtName = filename.substring(filename.lastIndexOf(".") + 1);
								fileBean.setExtName(fileExtName);
								// 如果需要限制上传的文件类型，那么可以通过文件的扩展名来判断上传的文件类型是否合法
								System.out.println("上传的文件的扩展名是：" + fileExtName);
								// 获取item中的上传文件的输入流
								InputStream in = item.getInputStream();
								// 得到文件保存的名称
								String saveFilename = makeFileName(filename);
								// 得到文件的保存目录
								String dbDir = makePath(saveFilename);
								String realSavePath = savePath + dbDir;
								// File既可以代表文件也可以代表目录
								File file = new File(realSavePath);
								// 如果目录不存在
								if (!file.exists()) {
									// 创建目录
									file.mkdirs();
								}
								fileBean.setFieldName(filename);
								fileBean.setNewFileName(saveFilename);
								fileBean.setFilePath(dbDir + saveFilename);
								fileBeanList.add(fileBean);
								System.out.println(fileBean.getFilePath() + "============================");
								FileOutputStream out = new FileOutputStream(realSavePath + "\\" + saveFilename);
								byte buffer[] = new byte[1024];
								int len = 0;
								while ((len = in.read(buffer)) > 0) {
									out.write(buffer, 0, len);
								}
								in.close();
								out.close();
								// item.delete();
								message = "文件上传成功！";
							}
						}
					
	}
	
	
	private String makeFileName(String filename) { // 2.jpg
		// 为防止文件覆盖的现象发生，要为上传文件产生一个唯一的文件名
		return UUID.randomUUID().toString() + "_" + filename;
	}

	private String makePath(String filename) {
		// 得到文件名的hashCode的值，得到的就是filename这个字符串对象在内存中的地址
		int hashcode = filename.hashCode();
		int dir1 = hashcode & 0xf; // 0--15
		int dir2 = (hashcode & 0xf0) >> 4; // 0-15
		// 构造新的保存目录
		String dir = "/" + dir1 + "/" + dir2 + "/" ; // upload\2\3
															// upload\3\5
		return dir;
	}
	
	public HttpServletRequest getRequest() {
		return request;
	}
	public void setRequest(HttpServletRequest request) {
		this.request = request;
	}
	public String getSavePath() {
		return savePath;
	}
	public void setSavePath(String savePath) {
		this.savePath = savePath;
	}
	public List<FileBean> getFileBeanList() {
		return fileBeanList;
	}
	public void setFileBeanList(List<FileBean> fileBeanList) {
		this.fileBeanList = fileBeanList;
	}
	public Map<String, String> getFormDatas() {
		return formDatas;
	}
	public void setFormDatas(Map<String, String> formDatas) {
		this.formDatas = formDatas;
	}
	public String getUploadStatus() {
		return uploadStatus;
	}
	public void setUploadStatus(String uploadStatus) {
		this.uploadStatus = uploadStatus;
	}
	public long getMaxSize() {
		return maxSize;
	}
	public void setMaxSize(long maxSize) {
		this.maxSize = maxSize;
	}
	public String getAllowExts() {
		return allowExts;
	}
	public void setAllowExts(String allowExts) {
		this.allowExts = allowExts;
	}
	public String getUnAllowExts() {
		return unAllowExts;
	}
	public void setUnAllowExts(String unAllowExts) {
		this.unAllowExts = unAllowExts;
	}
	public String getTempPath() {
		return tempPath;
	}
	public void setTempPath(String tempPath) {
		this.tempPath = tempPath;
	}
	public String getMessage() {
		return message;
	}

	public void setMessage(String message) {
		this.message = message;
	}
}
